import argparse
import os

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--api', type=str)
    parser.add_argument('--input_dir', type=str)
    parser.add_argument('--output_dir', type=str)
    args = parser.parse_args()

    if not os.path.exists(args.output_dir):
        os.makedirs(args.output_dir)

    for file in os.listdir(args.input_dir):
        input_path = os.path.join(args.input_dir, file)
        output_path = os.path.join(args.output_dir, file)
        os.system(
            f'python run_pt_api.py '
            f'--api={args.api} '
            f'--input_path={input_path} '
            f'--output_path={output_path}'
        )
